package org.example.tencent_translation;

import com.google.gson.Gson;
import com.tencentcloudapi.common.AbstractModel;
import com.tencentcloudapi.common.Credential;
import com.tencentcloudapi.common.exception.TencentCloudSDKException;
import com.tencentcloudapi.common.profile.ClientProfile;
import com.tencentcloudapi.common.profile.HttpProfile;
import com.tencentcloudapi.tmt.v20180321.TmtClient;
import com.tencentcloudapi.tmt.v20180321.models.TextTranslateRequest;
import com.tencentcloudapi.tmt.v20180321.models.TextTranslateResponse;
import org.example.translation.*;

import java.util.Map;

public class TencentTranslation extends DefaultLangSupportHandler {

    private final TmtClient client;
    private final Gson gson = new Gson();

    public TencentTranslation(PropValueSupplier propValueSupplier) {
        super(propValueSupplier);
        translationApiType = TranslationApiType.Interval;
        // 实例化一个认证对象，入参需要传入腾讯云账户 SecretId 和 SecretKey，此处还需注意密钥对的保密
        // 代码泄露可能会导致 SecretId 和 SecretKey 泄露，并威胁账号下所有资源的安全性。以下代码示例仅供参考，建议采用更安全的方式来使用密钥，请参见：https://cloud.tencent.com/document/product/1278/85305
        // 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
        Credential cred = new Credential(propValueSupplier.get("tencent.secretId"), propValueSupplier.get("tencent.secretKey"));
        // 实例化一个http选项，可选的，没有特殊需求可以跳过
        HttpProfile httpProfile = new HttpProfile();
        httpProfile.setEndpoint("tmt.tencentcloudapi.com");
        // 实例化一个client选项，可选的，没有特殊需求可以跳过
        ClientProfile clientProfile = new ClientProfile();
        clientProfile.setHttpProfile(httpProfile);
        // 实例化要请求产品的client对象,clientProfile是可选的
        client = new TmtClient(cred, "ap-guangzhou", clientProfile);
    }

    @Override
    public TranslationResponseBean requestTranslation(TranslationPostBean postBean) {
        try {
            // 实例化一个请求对象,每个接口都会对应一个request对象
            TextTranslateRequest req = new TextTranslateRequest();
            req.setSourceText(postBean.getSourceString());
            req.setSource(getSupportLang(postBean.getSourceCountry()));
            req.setTarget(getSupportLang(postBean.getTargetCountry()));
            req.setProjectId(0L);
            // 返回的resp是一个TextTranslateResponse的实例，与请求对象对应
            TextTranslateResponse resp = client.TextTranslate(req);
            if (translationApiType == TranslationApiType.Interval){
                Thread.sleep(200);
            }
            // 输出json格式的字符串回包
            TextTransBean textTransBean = gson.fromJson(AbstractModel.toJsonString(resp), TextTransBean.class);
            if (textTransBean != null) {
                return new TranslationResponseBean()
                        .setSourceCountry(postBean.getSourceCountry())
                        .setSourceString(postBean.getSourceString())
                        .setTargetCountry(postBean.getTargetCountry())
                        .setTargetResult(textTransBean.getTargetText());
            }
        } catch (TencentCloudSDKException | InterruptedException e) {
            e.printStackTrace();
        }
        return new TranslationResponseBean()
                .setSourceCountry(postBean.getSourceCountry())
                .setSourceString(postBean.getSourceString())
                .setTargetCountry(postBean.getTargetCountry())
                .setTargetResult("未完成翻译");
    }
}
